# -*- coding: utf-8 -*-
import requests
import time
import random
import get_proxy_ip
import pymongo
import re
from apscheduler.schedulers.blocking import BlockingScheduler
from mongo_config import *

mongo_client = pymongo.MongoClient(MONGO_URL_TARGET, 27017)
db = mongo_client[MONGO_DB_RES]
collection = db['THS']

def get_html(url, header, proxy_ip):
    result_list = []
    print(url, header, proxy_ip)
    response = requests.get(url, headers = header, proxies = proxy_ip)
    text = response.text.split('item:[')[1].split(']};')[0]
    text_list = text.split('},')
    # print(text_list)
    pattern_time = re.compile(r'"pubDate":"(.*?)",', re.I)
    pattern_url = re.compile(r'"url":"(.*?)",', re.I)
    pattern_title = re.compile(r'"title":"(.*?)",', re.I)
    pattern_content = re.compile(r'"content":"(.*?)",', re.I)
    for text in text_list:
        dicts = {}
        # print(text)
        pub_date = pattern_time.findall(text)
        news_url = pattern_url.findall(text)
        title = pattern_title.findall(text)
        content = pattern_content.findall(text)
        # print(pub_date, news_url, title, content)
        dicts['_id'] = int(time.mktime(time.strptime(pub_date[0], "%Y/%m/%d %H:%M")))
        dicts['news_url'] = news_url[0]
        dicts['title'] = title[0]
        dicts['content'] = content[0]
        result_list.append(dicts)
    return result_list

def file_proxy_ip():
    pro_list = []
    with open('proxy_ip.txt', 'r') as f:
        r = f.read()
        r_list = r.split('\n')
        for r_t in r_list:
            pro_list.append(r_t)
        for i in range(len(pro_list)):
            ip = random.choice(pro_list)
            print(ip)
            pro_ip = 'http://' + ip.split('//')[1].split('}')[0][:-1]
            proxy_ip = {'http:': pro_ip}
            flag = verification_pro_ip(proxy_ip)
            if flag == True:
                return proxy_ip

def verification_pro_ip(pro_ip):
    url = 'https://www.baidu.com/'
    header = {'User-Agent': 'Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0'}
    try:
        code = requests.get(url, headers=header, proxies=pro_ip, timeout=10).status_code
        if code == 200:
            return True
        else:
            return False
    except:
        print('connect failed')

def save_to_mongo(results):
    # for result in results:
    #     print(result)
    try:
        collection.insert_many(results, ordered=False)
        print('insert ok!')
    except Exception as e:
        print(e)

def main():
    news_time = int(time.time() * 1000)
    user_agents = ['Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0',
                   'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)',
                   'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)',
                   'Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6; rv:2.0.1) Gecko/20100101 Firefox/4.0.1',
                   'Mozilla/5.0 (Windows NT 6.1; rv:2.0.1) Gecko/20100101 Firefox/4.0.1',
                   'Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; en) Presto/2.8.131 Version/11.11',
                   'Opera/9.80 (Windows NT 6.1; U; en) Presto/2.8.131 Version/11.11',
                   'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Maxthon 2.0)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; TencentTraveler 4.0)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; The World)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SE 2.X MetaSr 1.0; SE 2.X MetaSr 1.0; .NET CLR 2.0.50727; SE 2.X MetaSr 1.0)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; 360SE)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Avant Browser)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)',
                   'Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5',
                   'Mozilla/5.0 (iPod; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5',
                   'Mozilla/5.0 (iPad; U; CPU OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5',
                   'Mozilla/5.0 (Linux; U; Android 2.3.7; en-us; Nexus One Build/FRF91) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1',
                   'MQQBrowser/26 Mozilla/5.0 (Linux; U; Android 2.3.7; zh-cn; MB200 Build/GRJ22; CyanogenMod-7) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1',
                   'Opera/9.80 (Android 2.3.4; Linux; Opera Mobi/build-1107180945; U; en-GB) Presto/2.8.149 Version/11.10',
                   'Mozilla/5.0 (Linux; U; Android 3.0; en-us; Xoom Build/HRI39) AppleWebKit/534.13 (KHTML, like Gecko) Version/4.0 Safari/534.13',
                   'Mozilla/5.0 (BlackBerry; U; BlackBerry 9800; en) AppleWebKit/534.1+ (KHTML, like Gecko) Version/6.0.0.337 Mobile Safari/534.1+',
                   'Mozilla/5.0 (hp-tablet; Linux; hpwOS/3.0.0; U; en-US) AppleWebKit/534.6 (KHTML, like Gecko) wOSBrowser/233.70 Safari/534.6 TouchPad/1.0',
                   'Mozilla/5.0 (SymbianOS/9.4; Series60/5.0 NokiaN97-1/20.0.019; Profile/MIDP-2.1 Configuration/CLDC-1.1) AppleWebKit/525 (KHTML, like Gecko) BrowserNG/7.1.18124',
                   'Mozilla/5.0 (compatible; MSIE 9.0; Windows Phone OS 7.5; Trident/5.0; IEMobile/9.0; HTC; Titan)',
                   'UCWEB7.0.2.37/28/999',
                   'NOKIA5700/ UCWEB7.0.2.37/28/999',
                   'Openwave/ UCWEB7.0.2.37/28/999',
                   'Mozilla/4.0 (compatible; MSIE 6.0; ) Opera/UCWEB7.0.2.37/28/999',
                   'Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; InfoPath.2; .NET4.0C; .NET4.0E; .NET CLR 2.0.50727; 360SE)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SE 2.X MetaSr 1.0; SE 2.X MetaSr 1.0; .NET CLR 2.0.50727; SE 2.X MetaSr 1.0)',
                   'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11',
                   'Mozilla/5.0 (Linux; U; Android 2.2.1; zh-cn; HTC_Wildfire_A3333 Build/FRG83D) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1',
                   'Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_6_8; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50',
                   'Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; TencentTraveler 4.0; .NET CLR 2.0.50727)',
                   'MQQBrowser/26 Mozilla/5.0 (Linux; U; Android 2.3.7; zh-cn; MB200 Build/GRJ22; CyanogenMod-7) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1',
                   'Mozilla/5.0 (Windows NT 6.1; rv:2.0.1) Gecko/20100101 Firefox/4.0.1',
                   'Mozilla/5.0 (Androdi; Linux armv7l; rv:5.0) Gecko/ Firefox/5.0 fennec/5.0',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; The World)',
                   'Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Maxthon 2.0)',
                   'Opera/9.80 (Windows NT 6.1; U; en) Presto/2.8.131 Version/11.11',
                   'Opera/9.80 (Android 2.3.4; Linux; Opera mobi/adr-1107051709; U; zh-cn) Presto/2.8.149 Version/11.10',
                   'UCWEB7.0.2.37/28/999',
                   'NOKIA5700/ UCWEB7.0.2.37/28/999',
                   'Openwave/ UCWEB7.0.2.37/28/999',
                   'Mozilla/4.0 (compatible; MSIE 6.0; ) Opera/UCWEB7.0.2.37/28/999', ]
    proxy_ip = None
    try:
        proxy_ip = file_proxy_ip()
    except Exception as e:
        print(e)
    header = {'User-Agent': random.choice(user_agents)}
    news_url = 'http://stock.10jqka.com.cn/thsgd/realtimenews.js?_=' + str(news_time)
    results = get_html(news_url, header, proxy_ip)
    save_to_mongo(results)

if __name__ == '__main__':
    # main()
    # 定义BlockingScheduler
    sched = BlockingScheduler()
    sched.add_job(main, 'interval', minutes=5)  # minutes
    sched.add_job(get_proxy_ip.save_to_txt, 'interval', hours=4)  # hours, minutes
    sched.start()